import json
from typing import Any
from mcp.server.fastmcp import FastMCP
from datetime import datetime
mcp = FastMCP("DemoServer")

# 模拟调用
@mcp.tool()
async def get_weather(city:str)-> str:
    """
    输入指定城市名称，返回今日天气情况
    :param city:城市名称
    :return: 格式化后的天气情况
    """
    return (
        f"城市, {city}\n"
        f"温度: 20°C\n"
        f"湿度: 50%\n"
        f"风速: 10 m/s\n"
        f"天气: 小雨\n"
    )


@mcp.tool()
async def get_now() -> str:
    """
    获取当前时间
    :return: 格式化后的时间
    """
    current_datatime = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    return f"当前时间: {current_datatime}"


if __name__ == "__main__":
    # 以标准I/O方式运行 MCP 服务器
    mcp.run(transport='stdio')